#include <iostream>
#include <memory>
#include "UdpServer.hpp"
#include "Route.hpp"
#include "ThreadPool.hpp"
// std::string defaulthandler(const std::string &message)
// {
//     std::string hello = "hello, ";
//     hello += message;
//     return hello;
// }

using namespace ThreadPoolModule;

using task_t = std::function<void()>;

// ./udpserver ip + port
// ./udpserver port
int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << "ip port" << std::endl;
        return 1;
    }
    // std::string ip = argv[1];
    uint16_t port = std::stoi(argv[1]); // 字符串转整形

    Enable_Console_Log_Strategy(); // 开启日志

    // 1. 路由服务
    Route r;
    // 2. 线程池
    auto tp = ThreadPool<task_t>::GetInstance();
    // 3. 网络服务器对象，提供通信功能
    // 2. 网络服务器对象，提供通信功能
    // std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port);
    // std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r](int sockfd, const std::string &message, InetAddr &peer)
    // { r.MessageRoute(sockfd, message, peer); });
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr &peer)
    { auto t = std::bind(&Route::MessageRoute, &r, sockfd, message, peer);
                        //使用r对象的MessageRoute方法, sockfd, message, peer引用的参数
    tp->Enqueue(t); });
    usvr->Init();
    usvr->Start();
    return 0;
}